#!/bin/bash

stage=1

train_scp_path="data/train_combined_2000k/feats.scp"
train_utt2spk="data/train_combined_2000k/utt2spk"

val_scp_path="data/train_combined_5k/feats.scp"
val_utt2spk=$train_utt2spk

test_scp_path="data/train_combined_2000k/feats.scp"
test_utt2spk="data/train_combined_2000k/utt2spk"


if [ $stage -eq 1 ];then
	python -u main.py \
		--max_epochs 20 \
		--batch_size 1024 \
		--num_workers 64 \
		--save_top_k 5 \
		--train_scp_path $train_scp_path \
		--train_utt2spk $train_utt2spk \
		--val_scp_path $val_scp_path \
		--val_utt2spk $val_utt2spk \
		--test_scp_path $test_scp_path \
		--test_utt2spk $test_utt2spk \
		--learning_rate 1e-3 \
		--min-length 200 \
		--gpus 4 \
		--distributed_backend ddp
fi

